package summary;

import java.util.*;

/**
 * @Author: 海琳琦
 * @Date: 2022/6/9 14:28
 * https://leetcode.cn/problems/find-k-pairs-with-smallest-sums/
 */
public class Title373 {

    /**
     * 滑动窗口+优先队列
     * @param nums1
     * @param nums2
     * @param k
     * @return
     */
    public List<List<Integer>> kSmallestPairs(int[] nums1, int[] nums2, int k) {
        //构建小根堆（存放索引）
        PriorityQueue<int[]> queue = new PriorityQueue<>((a, b) -> ((nums1[a[0]] + nums2[a[1]]) - (nums1[b[0]] + nums2[b[1]])));
        for (int i = 0; i < nums1.length && i < k; i++) {
            //先放[0,0]，[1,0],[2,0],[3,0]
            queue.offer(new int[]{i, 0});
        }
        List<List<Integer>> result = new ArrayList<>();
        while (!queue.isEmpty() && k-- > 0) {
            int[] poll = queue.poll();
            result.add(Arrays.asList(nums1[poll[0]], nums2[poll[1]]));
            if (++poll[1] < nums2.length) {
                queue.offer(poll);
            }
        }
        return result;
    }

    public static void main(String[] args) {

    }
}
